/**
 * @file yolov5s_postprocess.h
 * @author your name (you@domain.com)
 * @brief 目标检测的后处理相关事务
 * @version 0.1
 * @date 2022-12-17
 * 
 * @copyright Copyright (c) 2022
 * 
 */

#ifndef YOLOV5S_POSTPROCESS_H
#define YOLOV5S_POSTPROCESS_H

//这里包含了yolov5检测所有需要的头文件
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <string.h>
#include <stdint.h>
#include <iostream>
#include <vector>
#include "rknn_api.h"
#include "customized_struct.h"

int post_process(uint8_t *input0, uint8_t *input1, uint8_t *input2, int model_in_h, int model_in_w,
                 float conf_threshold, float nms_threshold, float vis_threshold, float scale_w, float scale_h,
                 std::vector<uint8_t> &qnt_zps, std::vector<float> &qnt_scales,
                 detect_result_group_t *group,float h_offset);
#endif